package com.controller;

import com.util.UUIDUtils;
import org.apache.commons.io.FilenameUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * @author mornd
 * @date 2020/11/26 - 21:52
 */

/**
 * 通用文件上传控制器
 */
@RestController
@RequestMapping("/upload")
public class PublicUploadController {
    /**
     * 上传图片
     *
     * @param file           上传的富文本编辑器图片
     * @param attach         上传的普通文件对象
     * @param fileUploadPath 上传的路径
     * @return
     */
    @RequestMapping("/uploadImages")
    public Map<String, Object> uploadFile(MultipartFile file, MultipartFile attach, String fileUploadPath) {
        //如果不为空则表示上传的是房间详情图片
        if (file != null) attach = file;
        //返回至前台对应的数据格式
        Map<String, Object> maps = new HashMap<String, Object>();
        //判断是否有文件
        if (!attach.isEmpty()) {
            //获取原文件名称
            String oldName = attach.getOriginalFilename();
            //获取文件后缀名
            String extension = FilenameUtils.getExtension(oldName);
            //重命名文件名称
            String newFileName = UUIDUtils.randomID() + "." + extension;
            //新建文件夹解决文件夹过多问题
            String datePath = new SimpleDateFormat("yyyyMMdd").format(new Date());
            //拼接最后的文件名
            String finalName = datePath + "/" + newFileName;
            //创建文件对象
            File destFile = new File(fileUploadPath, finalName);
            //判断文件夹是否存在 不存在则创建文件夹
            if (!destFile.getParentFile().exists()) {
                destFile.getParentFile().mkdirs();
            }
            try {
                //保存文件至磁盘
                attach.transferTo(destFile);
                //保存相应的数据
                maps.put("code", 0);
                maps.put("msg", "");
                //房型详情的src默认是框架自己拿的 必须符合规范
                if (file != null) {
                    Map<String, Object> dataMaps = new HashMap<String, Object>();
                    dataMaps.put("src", "/hotel/show/roomImages/detail/" + finalName);//文件上传成功的回显地址
                    maps.put("data", dataMaps);
                }
                maps.put("imagePath", finalName);//图片名称，目的是给photo隐藏域赋值
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return maps;
    }
}
